package com.azxcrm.project.monitor.job.task;

import java.io.BufferedReader;
import java.io.FileOutputStream;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.text.SimpleDateFormat;
import java.util.Date;

import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;

/**
 * 每日定时调度器 之 数据库备份
 * 
 * @author Dustyone
 *
 */
@Component("dataBaseBackup")
public class DataBaseBackup {

	/**
	 * 执行的命令语句
	 */
	 @Value("${task.backup.command}")
	private String command;

	/**
	 * 编码方式
	 */
	 @Value("${task.backup.encode:'utf-8'}")
	 private String encode;

	/**
	 * 导出文件路径前缀
	 */
	 @Value("${task.backup.pre_fout:'f:\\test-'}")
	 private String pre_fout;
	 
	 /**
	 * 导出文件路径后缀
	 */
	 @Value("${task.backup.suf_fout:'.sql'}")
	 private String suf_fout;

	public void backup() {
			
		try {
			Runtime rt = Runtime.getRuntime();
			
			// 调用 调用mysql的安装目录的命令
			Process child = rt.exec(this.command);
			// 设置导出编码为utf-8。这里必须是utf-8
			// 把进程执行中的控制台输出信息写入.sql文件，即生成了备份文件。注：如果不对控制台信息进行读出，则会导致进程堵塞无法运行
			InputStream in = child.getInputStream();// 控制台的输出信息作为输入流

			InputStreamReader inputFile = new InputStreamReader(in, this.encode);
			// 设置输出流编码为utf-8。这里必须是utf-8，否则从流中读入的是乱码

			String inStr;
			StringBuffer sb = new StringBuffer("");
			String outStr;
			// 组合控制台输出信息字符串
			BufferedReader br = new BufferedReader(inputFile);
			while ((inStr = br.readLine()) != null) {
				sb.append(inStr + "\r\n");
			}
			outStr = sb.toString();

			// 要用来做导入用的sql目标文件：
			FileOutputStream fout = new FileOutputStream(this.pre_fout+new SimpleDateFormat("yyyy-MM-dd").format(new Date())+this.suf_fout);
			OutputStreamWriter writer = new OutputStreamWriter(fout, this.encode);
			writer.write(outStr);
			writer.flush();
			in.close();
			inputFile.close();
			br.close();
			writer.close();
			fout.close();
			System.out.println("保存成功");
		} catch (Exception e) {
			e.printStackTrace();
		}

	}

}
